package cn.assumejson.springframework.beans.context.support;

import cn.assumejson.springframework.beans.context.ApplicationContext;
import cn.assumejson.springframework.beans.context.ApplicationContextAware;
import cn.assumejson.springframework.beans.factory.BeansException;
import cn.assumejson.springframework.beans.factory.config.BeanPostProcessor;

/**
 * ApplicationContext类不能够在BeanFactory启动之前获取，必须在refresh之后才会有。
 * 所以可以通过后置处理器的方式将ApplicationContext 传递出来
 * <p>
 * 让继承自 ApplicationContextAware 的 Bean 对象都能感知所属的 ApplicationContext。
 * 这样一来才是真正的上下文对象了。
 *
 * @author Mr. Li
 * @version 1.0
 * @since 1.0 2021/12/1 12:57
 */
public class ApplicationContextAwareProcessor implements BeanPostProcessor {

    private ApplicationContext applicationContext;

    public ApplicationContextAwareProcessor(ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        // 感知 ApplicationAware
        if (bean instanceof ApplicationContextAware) {
            ((ApplicationContextAware) bean).setApplicationContext(applicationContext);
        }
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }
}
